//
//  NumericCoder.hpp
//  Clock Signal
//
//  Created by Thomas Harte on 17/01/2023.
//  Copyright © 2023 Thomas Harte. All rights reserved.
//

#pragma once

namespace Numeric {

/// Stores and retrieves an arbitrary number of fields into an int with arbitrary modulos.
///
/// E.g. NumericCoder<8, 3, 14> establishes an encoder and decoder for three fields,
/// the first is modulo 8, the second is modulo 3, the third is modulo 14.
///
/// NumericCoder<8, 3, 14>::encode<2>(v, 9) will mutate v so that the third field
/// (i.e. field 2) has value 9.
///
/// The first given field will occupy the least significant part of the target int; e.g.
/// a NumericCoder<3, 4, 6> with ::encode<0>(v, 2), ::encode<1>(v, 1) and
/// ::encode<2>(v, 5) will have a final value of:
///
///		[value] 2 +
///		[value] 1 * [product of previous field sizes] 3 +
///		5 * 12
///				= 65
///
template <int... Sizes> class NumericCoder {
public:
	/// Modifies @c target to hold @c value at @c index.
	template <int index> static void encode(int &target, const int value) {
		static_assert(index < sizeof...(Sizes), "Index must be within range");
		NumericEncoder<Sizes...>::template encode<index>(target, value);
	}

	/// @returns The value from @c source at @c index.
	template <int index> static int decode(const int source) {
		static_assert(index < sizeof...(Sizes), "Index must be within range");
		return NumericDecoder<Sizes...>::template decode<index>(source);
	}

private:

	template <int size, int... Tail>
	struct NumericEncoder {
		template <int index, int i = 0, int divider = 1> static void encode(int &target, const int value) {
			if constexpr (i == index) {
				const int suffix = target % divider;
				target /= divider;
				target -= target % size;
				target += value % size;
				target *= divider;
				target += suffix;
			} else {
				NumericEncoder<Tail...>::template encode<index, i+1, divider*size>(target, value);
			}
		}
	};

	template <int size, int... Tail>
	struct NumericDecoder {
		template <int index, int i = 0, int divider = 1> static int decode(const int source) {
			if constexpr (i == index) {
				return (source / divider) % size;
			} else {
				return NumericDecoder<Tail...>::template decode<index, i+1, divider*size>(source);
			}
		}
	};
};

}
